tensor([[1., 0., 1., 0., 1.],
        [0., 1., 0., 1., 0.]], grad_fn=<RoundBackward1>)